#第一个二级菜单
def getall():
    print()
    print(f'图书馆一共有{len(names)}本书')
    name_ls = list(names.keys())
    for count, i in enumerate(name_ls):
        print(f'{count + 1}.{i}---{names[i]}')
    print()
    return name_ls

def getmenu():
    getall()
    while 1:
        print('a.修改一本书\nb.删除一本图书\nc.回到上级菜单')
        # 获取用户输入
        op = input('请选择您想要进行的操作：')
        if op == 'a':
            # 调用修改的函数
            update()
        elif op == 'b':
            # 调用删除的函数
            delete()
        elif op == 'c':
            break
        else:
            print('没有这个选项。')

def delete():
    name_ls = getall()
    i = int(input('请输入要删除的序号：'))
    if i < 1 or i > len(name_ls):
        print('没这个人')
        return
    op = input(f'您要删除的是：【{name_ls[i - 1]}】,确定删除吗（y）：')
    if op == 'y':
        names.pop(name_ls[i - 1])
        print('删除成功！')
    else:
        print('删除已取消')
    getall()

#添加的函数
def add():
    while 1:
        name = input('请输入要添加的图书（按回车退出）：')
        # 创建一个列表，把用户输入的信息都放进去
        if name == '':
            getall()
            break
        gender = input('请输入要添加图书的信息（按回车退出）：')
        # 把这个列表加入上面那个列表中
        names[name] = gender

# 修改的函数
def update():
    name_ls = getall()
    i = int(input('请输入要修改的图书序号：'))
    if i < 1 or i > len(name_ls):
        print('没这本书')
        return
    print(f'您输入的序号原来是：{name_ls[i - 1]}')
    na = input('请输入新的图书名字：')
    info = input('请输入新的图书信息：')
    names.pop(name_ls[i - 1])
    names[na] = info
    print('修改成功！')
    getall()

# 清除
def clear():
    names.clear()
    print('已清除所有图书')
    getall()

if __name__ == "__main__":
    # 用于存储所有的名单
    names = {"简爱": "莎士比亚",
             "牛逼": "龙傲天",
             "西游记": "吴承恩",
             "红楼梦": "曹雪芹"}
    while True:
        print('''
    =======图书馆书单======
    1.查看所有图书
    2.添加图书
    3.清除所有图书
    4.退出系统
        ''')
        # 获取用户的选择
        op = input('请选择：')
        if op == '1':
            # 第一个二级菜单
            getmenu()
        # 增加
        elif op == '2':
            add()
        elif op == '3':
            clear()
        elif op == '4':
            print('欢迎下次使用图书管理系统！')
            break
        else:
            print('没有这个选项。')